genetic algorithm python code